package com.faker.core.jdbc;

/**
 * 动态切换数据源数据源存储
 */
public class DataSourceContextHolder {
    // 使用 ThreadLocal 来存储当前线程持有的数据源类型
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    // 设置当前线程持有的数据源
    public static synchronized void setDataSourceType(String dataSourceType) {
        contextHolder.set(dataSourceType);
    }

    // 获取当前线程持有的数据源
    public static String getDataSourceType() {
        return contextHolder.get();
    }

    // 清除当前线程持有的数据源
    public static void clearDataSourceType() {
        contextHolder.remove();
    }
}